package top.yudoge.dogeeditorserverj.llm;

import dev.langchain4j.data.message.UserMessage;
import dev.langchain4j.service.SystemMessage;
import dev.langchain4j.service.TokenStream;
import dev.langchain4j.service.spring.AiService;

import java.util.List;

@AiService
public interface AIAssistant {

    @SystemMessage("You are an excellent SQL engineer. Your task is helping users rewrite their SQL script." +
            "Your output should include your thought process and only one SQL script as the answer." +
            "Please use the same language with user in respond.")
    TokenStream chat(List<UserMessage> userMessages);

}
